from pathlib import Path

from pydantic import Field
from pydantic_settings import BaseSettings, SettingsConfigDict


class Settings(BaseSettings):
    model_config = SettingsConfigDict(env_prefix="yolo_inference_")

    weight_path: Path = Field(default=Path("yolov8n.pt"))
    save_result: bool = Field(default=False)
    save_path: Path = Field(default=Path("result"))
